import argparse


def get_parser_args():
    """
    Parses the node2vec arguments.
    """
    parser = argparse.ArgumentParser(description="Metapath2vec")
    # parser.add_argument('--input_file', type=str, help="input_file")
    parser.add_argument('--aminer', action='store_true', help='Use AMiner dataset')
    # 输入文件
    parser.add_argument('--path', type=str, help="input_path", default='../../data/net_dbis/output_path.txt')
    # 输出文件
    parser.add_argument('--output_file', type=str, help='output_file', default='result.txt')
    # embedding维度
    parser.add_argument('--dim', default=128, type=int, help="embedding dimensions")
    # 窗口大小
    parser.add_argument('--window_size', default=7, type=int, help="context window size")
    # 迭代次数
    parser.add_argument('--iterations', default=5, type=int, help="iterations")
    # batch size
    parser.add_argument('--batch_size', default=50, type=int, help="batch size")
    # 0: metapath2vec; 1: metapath2vec++
    parser.add_argument('--care_type', default=0, type=int,
                        help="if 1, heterogeneous negative sampling, else normal negative sampling")
    # 学习率
    parser.add_argument('--initial_lr', default=0.025, type=float, help="learning rate")
    # skip2gram模型的词频
    parser.add_argument('--min_count', default=5, type=int, help="min count")
    # 资源核数
    parser.add_argument('--num_workers', default=16, type=int, help="number of workers")
    parser.set_defaults(directed=False)

    return parser.parse_args()
